//
// This file is part of the aMule Project.
//
// Copyright (c) 2004-2006 Angel Vidal (Kry) ( kry@amule.org )
// Copyright (c) 2004-2006 aMule Team ( admin@amule.org / http://www.amule.org )
// Copyright (c) 2003 Barry Dunne (http://www.emule-project.net)
//
// Any parts of this program derived from the xMule, lMule or eMule project,
// or contributed by third-party developers are copyrighted by their
// respective authors.
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
// 
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301, USA
//

// Note To Mods //
/*
Please do not change anything here and release it..
There is going to be a new forum created just for the Kademlia side of the client..
If you feel there is an error or a way to improve something, please
post it in the forum first and let us look at it.. If it is a real improvement,
it will be added to the offical client.. Changing something without knowing
what all it does can cause great harm to the network if released in mass form..
Any mod that changes anything within the Kademlia side will not be allowed to advertise
there client on the eMule forum..
*/

#ifndef __KAD_KADEMLIA_H__
#define __KAD_KADEMLIA_H__

#include <map>
#include "Types.h"
#include "UInt128.h"
#include "SearchManager.h"
#include "NoteIndex.h"

extern "C" {
#include "net.h"
}
class CSharedFileList;
struct Status;

////////////////////////////////////////
namespace std {
////////////////////////////////////////

class CPrefs;
class CRoutingZone;
class CKademliaUDPListener;
class CKademliaError;
class CSearch;
class CSearchManager;
class CContact;
//class CIndexed;
class CEntry;

typedef std::map<CRoutingZone*, CRoutingZone*> EventMap;

#define KADEMLIA_VERSION 0.1

class CKademlia
{
public:
        void start(char *arg1,char *arg2, int hashtype);//, void* curNodeMem);
        void start(CPrefs *prefs, char *arg1, char *arg2);//, void* curNodeMem);
	void stop();
	
	CPrefs				*getPrefs(void);
	CRoutingZone			*getRoutingZone(void);
	CKademliaUDPListener	*getUDPListener(void);
	//static CIndexed				*getIndexed(void);
	bool					isRunning(void) {return m_running;}
	bool					isConnected(void);
	bool					isFirewalled(void);
	void					RecheckFirewalled(void);
	uint32				getKademliaUsers(void);
	uint32				getKademliaFiles(void);
	uint32				getTotalStoreKey(void);
	uint32				getTotalStoreSrc(void);
	uint32				getTotalStoreNotes(void);
	uint32				getTotalFile(void);
	bool					getPublish(void);
	uint32				getIPAddress(void);
	void					bootstrap(uint32 ip, uint16 port);
	void					processPacket(const byte* data, uint32 lenData, uint32 ip, uint16 port);
	void					sendPacket(unsigned char* data, uint32 lenData, uint32 remoteip, uint16 port);
	void addEvent(CRoutingZone *zone);
	void removeEvent(CRoutingZone *zone);
	void process();//time_t now);
	bool string2int128(unsigned char *i128, char *s);
	UDPIO				*getUDPIO(void);
	bool getPacket();
	char *h_to_a(unsigned long int ip);
	
	unsigned long int get_bytes_sent(void) { return bytes_sent; }
	int get_packet_sent(void) { return packets_sent; }
	void clear_bytes_sent(void) { bytes_sent = 0; }
	void clear_packets_sent(void) { packets_sent = 0; }
	void restoreDVNState(void* curNodeMem);
	void saveDVNState(void* curNodeMem);

	//[dfk]  Maybe all the initialization should be done here
	CKademlia() {}
  ~CKademlia();
	
//private:

	//The modified code.  Each instance of a CKademlia object will now have its own copy of the variables
	CKademlia *instance;
	EventMap	m_events;
	time_t	m_nextSearchJumpStart;
	time_t	m_nextSelfLookup;
	time_t	m_nextFirewallCheck;
	time_t	m_nextFindBuddy;
	time_t	m_statusUpdate;
	time_t	m_bigTimer;
	time_t	m_bootstrap;
	bool		m_running;
	unsigned long int bytes_sent;
	int packets_sent;
	int sleep_until;// Time in seconds until the next event
	time_t m_nextRoutingTableDump;
	
	
	CPrefs					*m_prefs;
	CRoutingZone				*m_routingZone;
	CKademliaUDPListener			*m_udpListener;
	//CIndexed				*m_indexed;
	UDPIO 					*myUDP;
	CSearchManager                          *mySearchMan;
  NoteIndex myNoteIndex;
};

} // End namespace

#endif // __KAD_KADEMLIA_H__














